import os

default_bert_path = r'F:\pre_trained_model\chinese_L-12_H-768_A-12\chinese_L-12_H-768_A-12'
default_gpt_path = r'F:\pre_trained_model\GPT_LCCC-large-tf'


class BertInfo():
    def __init__(self,bert_model_path=default_bert_path):
        self.config_path = os.path.join(bert_model_path,'bert_config.json')
        self.checkpoint_path = os.path.join(bert_model_path,'bert_model.ckpt')
        self.dict_path = os.path.join(bert_model_path,'vocab.txt')

class GptInfo():
    def __init__(self,gpt_model_path = default_gpt_path):
        self.config_path = os.path.join(gpt_model_path, 'gpt_config.json')
        self.checkpoint_path = os.path.join(gpt_model_path, 'gpt_model.ckpt')
        self.dict_path = os.path.join(gpt_model_path, 'vocab.txt')

